package com.awenhui.demo.core.utils.zutils;

/**
 * zip处理工具类
 *
 * @author zhouxuanhong
 * @date 2019-10-28 21:16
 * @since JDK1.8
 */

import java.io.*;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;

public class ZipCompress {

    /**
     * @param zipFileName    ${@link String}目的地Zip文件
     * @param sourceFileName ${@link String}源文件（待压缩的文件或文件夹）
     * @author zhouxuanhong
     * @date 2019-10-28 21:18
     */
    public static void zip(String zipFileName, String sourceFileName) throws Exception {
        //创建zip输出流
        ZipOutputStream out = new ZipOutputStream(new FileOutputStream(zipFileName));

        //创建缓冲输出流
        BufferedOutputStream bos = new BufferedOutputStream(out);

        File sourceFile = new File(sourceFileName);

        //调用函数
        compress(out, sourceFile, sourceFile.getName());

        bos.close();
        out.close();
    }

    /**
     * 创建zip
     *
     * @param out        ${@link ZipOutputStream}
     * @param sourceFile ${@link File}
     * @param base       ${@link String}
     * @author zhouxuanhong
     * @date 2019-10-28 21:18
     */
    private static void compress(ZipOutputStream out, File sourceFile, String base) throws Exception {
        //如果路径为目录（文件夹）
        if (sourceFile.isDirectory()) {
            //取出文件夹中的文件（或子文件夹）
            File[] files = sourceFile.listFiles();
            //如果文件夹为空，则只需在目的地zip文件中写入一个目录进入点
            if (files.length == 0) {
                out.putNextEntry(new ZipEntry(base + "/"));
            } else {//如果文件夹不为空，则递归调用compress，文件夹中的每一个文件（或文件夹）进行压缩
                for (int i = 0; i < files.length; i++) {
                    compress(out, files[i], base + "/" + files[i].getName());
                }
            }
        } else {//如果不是目录（文件夹），即为文件，则先写入目录进入点，之后将文件写入zip文件中
            out.putNextEntry(new ZipEntry(base));
            FileInputStream fos = new FileInputStream(sourceFile);
            BufferedInputStream bis = new BufferedInputStream(fos);
            int tag;
            //将源文件写入到zip文件中
            while ((tag = bis.read()) != -1) {
                out.write(tag);
            }
            bis.close();
            fos.close();
        }
    }
}
